package com.breeze.leetcode;

import java.util.ArrayList;
import java.util.Arrays;

/**
 * https://leetcode-cn.com/problems/product-of-array-except-self/
 * 238. 除自身以外数组的乘积
 * 给你一个长度为 n 的整数数组 nums，其中 n > 1，返回输出数组 output ，其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
 * <p>
 * 提示：题目数据保证数组之中任意元素的全部前缀元素和后缀（甚至是整个数组）的乘积都在 32 位整数范围内。
 * <p>
 * 说明: 请不要使用除法，且在 O(n) 时间复杂度内完成此题。
 * <p>
 * 进阶：
 * 你可以在常数空间复杂度内完成这个题目吗？（ 出于对空间复杂度分析的目的，输出数组不被视为额外空间。）
 */
public class LeetCode238 {
    /**
     * 错误解答：全部相乘/当前数   原因：没有考虑除0问题,题目也要求不使用除法
     *
     * @param nums
     * @return
     */
    public int[] productExceptSelf(int[] nums) {
        int product = 1;
        for (int num : nums) {
            product *= num;
        }
        int[] output = new int[nums.length];
        for (int i = 0; i < output.length; i++) {
            output[i] = product / nums[i];
        }
        return output;
    }

    /**
     * 时间复杂度：O(n^2) 超出时间限制
     *
     * @param nums
     * @return
     */
    public int[] productExceptSelf1(int[] nums) {
        int[] output = new int[nums.length];
        Arrays.fill(output, 1);
        for (int i = 0; i < nums.length; i++) {
            for (int j = 0; j < output.length; j++) {
                if (i != j) {
                    output[j] *= nums[i];
                }
            }
        }
        return output;
    }

    /**
     * 时间复杂度：O(n)
     * 空间复杂度：O(1)
     */
    public int[] productExceptSelf2(int[] nums) {
        int[] output = new int[nums.length];
        // 先得到当前数前半部分的乘积
        int left = 1;
        for (int i = 0; i < nums.length; i++) {
            output[i] = left;
            left *= nums[i];
        }

        // 再得到当前数后半部分的乘积
        int right = 1;
        for (int i = nums.length - 1; i >= 0; i--) {
            output[i] *= right;
            right *= nums[i];
        }
        return output;
    }


    public static void main(String[] args) {
        LeetCode238 leetCode238 = new LeetCode238();
        int[] nums = {1, 2, 3, 4};
        int[] res = leetCode238.productExceptSelf2(nums);
        for (int re : res) {
            System.out.print(re + ",");
        }
    }
}
